package org.dreamwork.cai.monitor.web;

import org.dreamwork.cai.monitor.ws.MonitorWebSocket;
import org.dreamwork.websocket.WebSocketScanner;

import javax.websocket.Endpoint;
import java.util.HashSet;
import java.util.Set;

public class WebSocketConfig extends WebSocketScanner {
    private final Set<Class<? extends Endpoint>> classes = new HashSet<> ();

    public WebSocketConfig () {
        classes.add (MonitorWebSocket.class);
    }

    @Override
    protected Set<Class<? extends Endpoint>> getSupportedEndpointClasses () {
        return classes;
    }

    public Set<Class<? extends Endpoint>> getClasses () {
        return classes;
    }
}
